package com.feixiang.springbootinit.utils;

import com.feixiang.springbootinit.common.ErrorCode;
import com.feixiang.springbootinit.exception.BusinessException;
import lombok.extern.slf4j.Slf4j;

import java.util.List;
import java.util.function.Consumer;
@Slf4j
public class BatchSaveUtil {
    // 默认批次大小
    private static final int DEFAULT_BATCH_SIZE = 500;

    /**
     * 批量保存
     * @param list 要保存的数据列表
     * @param saveAction 保存操作（如service::saveBatch）
     * @param dataType 数据类型描述（用于日志，如"撤回消息"）
     * @param <T> 列表元素类型
     */
    public static <T> void batchSave(
            List<T> list,
            Consumer<List<T>> saveAction,
            String dataType
    ) {
        if (list == null || list.isEmpty()) {
            log.info("{}列表为空，无需保存", dataType);
            return;
        }

        try {
            for (int i = 0; i < list.size(); i += DEFAULT_BATCH_SIZE) {
                int end = Math.min(i + DEFAULT_BATCH_SIZE, list.size());
                saveAction.accept(list.subList(i, end));
            }
            log.info("成功保存 {} 条{}", list.size(), dataType);
        } catch (Exception e) {
            String errorMsg = "批量保存" + dataType + "失败";
            log.error(errorMsg, e);
            throw new BusinessException(ErrorCode.OPERATION_ERROR, errorMsg);
        }
    }
}
